package com.pg.agent.engine.service.component;

import com.pg.agent.engine.client.model.agent.AIAgent;
import lombok.Data;

/**
 * 任务组件
 * @author huhuaping
 * @date 2023/11/2 11:13
 */
@Data
public class TaskComponent extends PromptComponent {
    private String task;

    public TaskComponent(String task) {
        super();
        this.task = task;
    }

    @Override
    public String getPrompt(AIAgent agent) {
        return "你需要执行任务：" + task + "\n";
    }

    public String getTask() {
        return task;
    }

    public void setTask(String task) {
        this.task = task;
    }
}
